import databricks.koalas as ks
import numpy as np
import pandas as pd
import pyspark.pandas as ps

if __name__ == '__main__':
    kdf = ps.DataFrame({'A': ['foo', 'bar', 'foo', 'bar', 'foo', 'bar', 'foo', 'foo'],
                        'B': ['one', 'one', 'two', 'three', 'two', 'two', 'one', 'three'],
                        'C': np.random.randn(8),
                        'D': np.random.randn(8)})
    print("kdf索引", kdf.index)
    print("kdf列", kdf.columns)
    print("kdf头", kdf.head())
    print("kdf numpy", kdf.to_numpy())
    print("kdf描述", kdf.describe())
    # print("kdf转置", kdf.T)
